package com.yj.provider.dynamicDataSource;

/**
 * 定义数据源切换变量
 */
public class DataSourceContextHolder {

    /*
        数据源 basename 存放容器
     */
    private final static ThreadLocal<String> contextHolder = new ThreadLocal<>();

    /*
        设置数据源类型
     */
    public static void setContextHolder(String typeName){
        contextHolder.set(typeName);
    }

    /*
        获取数据源类型
     */
    public static String getDBType(){
        return contextHolder.get();
    }

    /*
        清楚数据源类型
     */
    public static void clearDBType(){
        contextHolder.remove();
    }

}
